Cleanup and add task_function.h to repo.