Edited to PEP8