Add time import

When using the -wait option the script fails because is missing the time import
Merge pull request #137 from fsuarez6/patch-1

Add time import