Can we make this code pep8? Spaces around the `=` for example?