Prime Number In Python Using Functions