1 files changed, 4 insertions, 4 deletions
diff --git a/lib/include/network.hpp b/lib/include/network.hpp
index 5cbba0d..e6d67b9 100644
--- a/lib/include/network.hpp
+++ b/lib/include/network.hpp
@@ -6,10 +6,11 @@
 #include <utility>
 #include <random>
 
-#include "cholmod.h"
+#include <cholmod.h>
 
 #include "graph.hpp"
 #include "hooks.hpp"
+#include "current_info.hpp"
 
 #ifdef FRACTURE_LONGINT
 
@@ -40,9 +41,8 @@ class network {
     ~network();
 
     void set_thresholds(double, std::mt19937&);
-    void break_edge(unsigned int);
-    void add_edge(unsigned int);
-    std::pair<double, std::vector<double>> currents();
+    void break_edge(unsigned int, bool unbreak = false);
+    current_info get_current_info();
     void fracture(hooks&, double cutoff = 1e-14);
 };