modified version of class diagram