Add arg passing to wrapper funcs from top level
Add initial state wfunc module