Fix application of coupling.
Add rhs for master equation.