diff --git a/main/main.pde b/main/main.pde
index 8ca9220..70562f3 100644
--- a/main/main.pde
+++ b/main/main.pde
@@ -1,4 +1,5 @@
 boolean pointclickedbool = false;
+boolean linedrawn = false;
 Point pointclicked;
 PointList list = new PointList();
 DrawHalfEdge drawgraph = new DrawHalfEdge();  
@@ -90,41 +91,31 @@ void draw() {
    drawgraph.drawGraph(list.getPoints());
 }
 void mouseReleased() {
-  Point p = new Point(mouseX,mouseY);
-  list.addPoint(p);
-}
-
-/*public void draw() {
-  if(mousePressed) {
-    float x = mouseX;
-    float y = mouseY;
-    if(pointclickedbool) {
-      pointclickedbool = false;
-      for(Point p: list.getPoints()) {
-        if(p.getX() < x-10 || p.getX() > x + 10)
-        {
-          if(p.getY() < y - 10 || p.getY() > y + 10) {
-            HalfEdge h = new HalfEdge(p, pointclicked);
-          }
+  float x = mouseX;
+  float y = mouseY;
+  if(pointclickedbool) {
+    pointclickedbool = false;
+    for(Point p: list.getPoints()) {
+      if((p.getX() - 5 < x) && (p.getX() + 5 > x)) {
+        if((p.getY() - 5 < y) && (p.getY() +5 > y)) {
+          HalfEdge h = new HalfEdge(p, pointclicked);
+          linedrawn = true;
         }
       }
-    } else {
-      for(Point p: list.getPoints()) {
-        if((p.getX() < x-10) && (p.getX() > x + 10))
-        {
-          if((p.getY() < y - 10) && (p.getY() > y + 10)) {
-            pointclickedbool = true;
-            pointclicked = p;
-          }
+    }
+  } else {
+    for(Point p: list.getPoints()) {
+      if((p.getX() - 5 < x) && (p.getX() + 5 > x)) {
+        if((p.getY() - 5 < y) && (p.getY() +5 > y)) {
+          pointclickedbool = true;
+          pointclicked = p;
         }
       }
     }
+  }
+  if(!pointclickedbool && !linedrawn) {
     Point p = new Point(x,y);
     list.addPoint(p);
-    if(pointclickedbool) {
-      System.out.println("true");
-    }
-    
-  }
-  draw.drawGraph(list.getPoints());
-}*/
+  }  
+  linedrawn = false;
+}