Cleaned code using flake8