fixed code with autopep8