Updated T2.fig and added get_prob function