Don't use sqrt.

Use a dictionary instead of a big if-else ladder.

Cracked it after 7 years :)