1 files changed, 8 insertions, 0 deletions
diff --git a/src/math/context.rs b/src/math/context.rs
index cb0bca0..e23b96f 100644
--- a/src/math/context.rs
+++ b/src/math/context.rs
@@ -58,6 +58,14 @@ impl Context {
     pub fn function(&self, name: &str) -> Option<&Arc<dyn Function>> {
         self.funcs.get(name)
     }
+
+    pub fn remove_variable(&mut self, name: &str) {
+        self.vars.remove(name);
+    }
+
+    pub fn remove_function(&mut self, name: &str) {
+        self.funcs.remove(name);
+    }
 }
 
 #[macro_export]