Optimisation

Optimisation of code