Adding solution to problem 8