HW1 part2 primes