Added solution to problem 21