Update code.js

Adding the 'highlighted' array to color some states.