PEP8

Cleaner code for a fantastic piece of python :)