working more on HW4