Add torch to build