Added color picker logic