Update problem1.js