pep8 compliance and included some testing