refactored tests
pep 8 fixes
respond to feedback