REF: now possible to input gradientStep