Added some PEP 8 compliance stuff