A Python script to generate maps of primes