Add class diagram homework 2