Adding bath_gd and batch_gd_stop version