Fixed some PEP 8 styling