updated version of class diagram