refactoring the code
code formatting, following PEP8 convention
We already have a `make style` command which automates formatting (with a setup that we chose).

Thanks for your contribution, closing this issue now.