Add tracking to predict and my_exp_plot