Taking out minor pep8 violations with the test.