Added macro to draw the oscillation probability