added a class diagram methingy