passing test & pep8
Merge pull request #10 from lauxley/master

flag-fix