add tests for task