Solution to problem 1 added