You are probably making an array upto n.
The tests on submit are usually bigger than the example tests.

Try to think of a better way to check if the number is prime.