added my partial solution for generating prime numbers in python