Tidied up for pep8
Tidied up for pep8