Added class diagram from the end of the project