Add torch overlayand remove un needed files