Addding class diagram