Adding solution to problem 1