Updated solution for problem 34