cleaned up code to to follow PEP8 guidelines.