added first version of the class diagram