Version of heat equation that works with units.