Add Torch into build