1 files changed, 12 insertions, 9 deletions
diff --git a/lib/include/wolff/graph.hpp b/lib/include/wolff/graph.hpp
index fcd73fa..65b3941 100644
--- a/lib/include/wolff/graph.hpp
+++ b/lib/include/wolff/graph.hpp
@@ -5,23 +5,26 @@
 #include <cmath>
 #include <vector>
 
-#include "types.h"
+namespace wolff {
 
-typedef enum lattice_t {
-  SQUARE_LATTICE,
-  DIAGONAL_LATTICE
-} lattice_t;
+#include "types.h"
 
-class graph_t {
+class graph {
   public:
+    D_t D;
+    L_t L;
     v_t ne;
     v_t nv;
-    std::vector<std::vector<v_t>> adj;
+    std::vector<std::vector<v_t>> adjacency;
     std::vector<std::vector<double>> coordinate;
 
-    graph_t(D_t D, L_t L, lattice_t lat = SQUARE_LATTICE);
-    void add_ext();
+    graph();
+    graph(D_t D, L_t L);
+
+    void add_ghost();
 };
 
+}
+
 #endif