play with adding in the gradient check conditions explicitly