Cleaning up whitespace with flake8
Making update from peer review