Fixed code to obey PEP8 rules