Corrected a typo in comment
Removed unused imports