Small fixes or optimisation