Take care of volume elements everywhere! Needs to be in all integrals (also t) and mass conservation check.