Add test cases for Tasks