add more tests, pep8-ify code