we need to redo the Class diagram