Add diagram for rest