hw1 prime number generation part