Adding problem 4