STY: Change code to be PEP8 compliant