add torch stuff