Implemented Unit3-HW1 as in solution video