Added class diagram and base class