Fix spin flips
Apply polarisation after each step to correctly calculate equation of
motion after spin flips.