add some comments and beautify code
comments & pylint :)