Update prime_time.py