Change to complay with PEP8