changed code style to conform to PEP 8