Fixed simple script
Fixed flake8 errors