pep8
fixed tests