Update module_prime.py