HW1 part2 prime numbers