Code is now PEP8 compliant