pep8
Adds basic tests