Update code-asg2.js