Adds corrections (i.e. gradient extrapolations) required for the GEAR-RT scheme during drifts, and some minor cleanup.
Here's some plots to show off the correction: