Python function to check if a number is prime or not

• Check if number is less than 2, return False

• Check if number is divisible by any number from 2 to its square root, return False

• Else, return True