Test a number to see if it's prime in python