add test script
clean up code, pep8