Split out assignment rewriting to own module.