2nd iteration of supervised (p=0.5, beta_2=0.25) imitation learning