minor update to task generator