Update in mainWFG

Forgot to remove old wfg(...) in mainWFG