X-Git-Url: http://nitlanguage.org

diff --git a/src/uml/uml_base.nit b/src/uml/uml_base.nit
index 3cb5e92..2580d95 100644
--- a/src/uml/uml_base.nit
+++ b/src/uml/uml_base.nit
@@ -16,23 +16,14 @@
 module uml_base
 
 import toolcontext
-import model
+import model::model_collect
 
-redef class ToolContext
-	# -p
-	var opt_privacy = new OptionBool("Generates private API", "-p", "--private")
-
-	# Shortcut for the value of `self.opt_privacy`
-	fun private_gen: Bool do return opt_privacy.value
+# UML model builder.
+class UMLModel
 
-	redef init do
-		option_context.add_option opt_privacy
-		super
-	end
-end
+	# Model view
+	var view: ModelView
 
-class UMLModel
-	var model: Model
+	# Main module used for linearization.
 	var mainmodule: MModule
-	var ctx: ToolContext
 end