Code to find if a number is prime