added comments in prime.py next_prime.py