Update problem4.js