add return stack to visualizatiobn
stack vis is basically done