Added class diagram. fixed access level of functions in figures