Adding solution for problem 8