Styling fixes per pep8