Add tests
pep8