Formatted the code according to PEP 8