Title: Optimal control of stochastic non-linear models