Almost done with HW2