Add in torch...

Why is this not in already?