Add: Io notch for function return. Also corrected shapes.