Addressing comments - pep8