Update problem3.1.js