Homework 1 fix