Adding class diagram...not sure if it's very good (or helpful) but at least we can say we have one