Changed code structure to follow PEP 8 guidelines