added color functionality