Forgot to import time module