Added support for autopep8. Works!
Fix whitespace and add some comments