Used autopep8 to cleanup python code style