add couple tests for task