unit tests finished and pep8 applied