Add task persistence