Add minor optimisation