Add beginnings of class diagram