Repaired the project, added two class diagrams