added torch test