fixup! Add diagram for rest