Update code_challenge.js