Code modified to keep to pep8 code standard