Template for problem 4