Re-added time module import. Oops!