Adding in Class diagrams for hierarchy persistence