Added torch function - just needs an appropriate icon with toggling