To speed up the function, she replaced the loop with a listcomp that filters and squares the numbers.

I refactored the module to replace several slow listcomps with generator expressions, which improved memory usage.