Optimising code for VEGA