Homework 2 in progress