Remove useless code + flake8