tests tests tests
adding more tests
flake8