Clean python code. Follow PEP8