First adding HW4 tasks 1,2