moved stochastic contribution to rhs function, rhs function should now return the state vector / state matrix at next time step. Added support for passing a user defined args to the stochastic rhs function