Remove torch for now