PEP8 Fixes

Modified the code to follow PEP8 standards.