more adagrad fixes