Fixed code to follow PEP8 conventions.