updated class diagram

not finished