Completed level 2, pep8 compliant