I think it would be a good thing to refactor the code to meet the [PEP 8](https://www.python.org/dev/peps/pep-0008/) guidelines.