Check if a number is prime

Define the function is_prime(n).

n is number that the function is_prime takes as a parameter.

The function is_prime return True if num is a prime number, False otherwise. Such as:

>>> from solution import is_prime
>>> is_prime(3)
True
>>> is_prime(4)
False