Added state visualization to the interface