Sme rhs restructuring

Changed behavior of RHS function in stochastic module.