Fixed PEP-8 voilations.

* add python headers.
* Remove extra whitespaces.