More code optimisations