Add partial redrawing of 2D visualization (#423)