Great, thank you! I'll try to come up with a faster way to do it.

There is a 6000ms time limit for the tests in order to test performance.
While your code works for smaller numbers like 5 & 10, it is probably too ineffecient for larger numbers.

In ruby for example, n may be as high as 10**8.
You need to think of a faster way to solve the problem.

I can't seem to get this past submission - it says "Process was terminated. It took longer than 6000ms to complete."

My test cases pass:
Test.assert_equals(locker_run(10), [1,4,9])
Test.assert_equals(locker_run(5),[1, 4])

Any suggestions/thoughts?