add torch back