Optimise code