Update problem2.js