updated class diagram

updated class diagram