Working on WGen for prototype