Update primes.py module

Added a few comments for readabiliry