updated class diagram
small fixes