pep8 compliance errors and added test