Add class diagram for Project 2