make autopep8 work better
formatting