add part of class diagram