change color scheme to match assertion graph