changed update to include color