[python] To find first N prime numbers in python