Prime Numbers kata for Python