Homework 1: pep8 fix