Fixed onColorChosen to handle parameter