Improve optimisation section