changing test to reflect latest updates + pep8 rules