Adding solution to problem 22