added HW 4 and 5