1 files changed, 6 insertions, 3 deletions
diff --git a/lib/include/wolff/graph.hpp b/lib/include/wolff/graph.hpp
index 165544a..0aeb6af 100644
--- a/lib/include/wolff/graph.hpp
+++ b/lib/include/wolff/graph.hpp
@@ -1,13 +1,16 @@
 
 #pragma once
 
-#include <inttypes.h>
 #include <cmath>
-#include <stdlib.h>
 #include <vector>
 
 #include "types.h"
 
+typedef enum lattice_t {
+  SQUARE_LATTICE,
+  DIAGONAL_LATTICE
+} lattice_t;
+
 class graph_t {
   public:
     v_t ne;
@@ -15,7 +18,7 @@ class graph_t {
     std::vector<std::vector<v_t>> v_adj;
     std::vector<std::vector<double>> coordinate;
 
-    graph_t(D_t D, L_t L);
+    graph_t(D_t D, L_t L, lattice_t lat = SQUARE_LATTICE);
     void add_ext();
 };