Add time import

When using the -wait option the script fails because is missing the time import