Made a public method private
Added some UML diagrams