Adding missing TimeUnit import