class diagram fixed