Add more tests for Task and TS and WF