update problem3 output