Add omnifunc support