more updates to HW 5