Sequence diagrams update

Added changes from TA review