Not a critical issue, but would be nice to setup and use `autopep8` for fixing style issues in all files.