Fixed some PEP-8 issues