is this an efficient way to check if a number is prime or not in python?