Unit test to run basic test, pep8 fixes