Add solution to problem set 2 and 3.