prime code snippets in python

check if a number is prime in python