Updated comment/description. Pep8