PEP8

Code is now PEP8 compliant