update to include color scheme