Fix Homework 4