Added color functionality