Added solution to problem 1