In that plot, the red curve is the x-momentum of the red (heavier) ball and the blue curve is for the yellow ball (yellow doesn't show up in the graph very well). The black line is the total momentum.