PEP8 Compliance

Updated code to comply with PEP8 style guidelines