Modelling the effects of the sawtooth instability in tokamaks using a current viscosity term