Added some comments
Removed incorrect and unused import