flag-fix

test passing & pep8