Optimisation UI + small bugfixes