add torch overlay back in