added comment on class diagram in code review