Optimised the RK integrator