Add TaskManagerContext class