add color-coded states