Created a class diagram to match