pep8 cleanup
added some basic tests