Homework4 in progress