Fixes/optimisations