Corrected typo
Removed the wait variable
Fixed import error