Formatting the code to PEP8