improved the visualization. decisionIdx is now treated correctly