Add colour highlighting for flx