Add import for time.sleep