Update code.js

JSHint pointed out some poor argument handling.