Added rough draft class diagram