added another class...

class diagrammey goodness