Add missing import for time module