More work on HW 4