add Torch back