Updated task docs and generator return