Add polling for completed tasks