Оформил код по pep8