Add solution for the 2nd homework