Add required time import statement to challenge_one.py