Added some readme comments
Pep8 & removing unused imports