Add missing import for time