ui fixes + optimisation