Add state visualization