Add event handling to handle color choosing