update template for problem 3