Add custom flag function to support both Go and Google flags