Add problem 4 of quiz