Cleaning up run_time.py