add color highlighting to state machine example