Update problem_1.lua