Autopep8 plus clean up