To change colorForCode, wouldn't it also need to take arguments for green and blue if it were to return the color in the 24-bit case?