diagram

updated class diagram