Updated code to meet PEP8 style guidelines.