diff --git a/js/display_atree.js b/js/display_atree.js
index 19e3f22..6e072cb 100644
--- a/js/display_atree.js
+++ b/js/display_atree.js
@@ -62,18 +62,10 @@ function construct_AT(elem, tree) {
                 document.getElementById("atree-row-" + i).children[node.display.col].appendChild(connector);
                 resolve_connector(document.getElementById("atree-row-" + i).children[node.display.col], node);
             }
-            // connect left
-            for (let i = parent_node.display.col + 1; i < node.display.col; i++) {
-                let connector = connect_elem.cloneNode()
-                connector.style.backgroundImage = "url('../media/atree/connect_line.png')";
-                connector.classList.add("rotate-90");
-                connector.id = "r" + parent_node.display.row + "-c" + i + "-line"
-                document.getElementById("atree-row-" + parent_node.display.row).children[i].appendChild(connector); 
-                resolve_connector(document.getElementById("atree-row-" + parent_node.display.row).children[i], node);
-            }
-
-            // connect right
-            for (let i = parent_node.display.col - 1; i > node.display.col; i--) {
+            // connect horizontally
+            let min = Math.min(parent_node.display.col, node.display.col);
+            let max = Math.max(parent_node.display.col, node.display.col);
+            for (let i = min + 1; i < max; i++) {
                 let connector = connect_elem.cloneNode()
                 connector.style.backgroundImage = "url('../media/atree/connect_line.png')";
                 connector.classList.add("rotate-90");
@@ -83,22 +75,19 @@ function construct_AT(elem, tree) {
             }
 
             // connect corners
-            if (parent_node.display.col > node.display.col && (parent_node.display.row != node.display.row)) {
-                let connector = connect_elem.cloneNode()
-                connector.style.backgroundImage = "url('../media/atree/connect_angle.png')";
-                connector.classList.add("rotate-180");
-                connector.id = "r" + parent_node.display.row + "-c" + node.display.col + "-angle"
-                document.getElementById("atree-row-" + parent_node.display.row).children[node.display.col].appendChild(connector);
-                resolve_connector(document.getElementById("atree-row-" + parent_node.display.row).children[node.display.col], node);
-            }
 
-            if (parent_node.display.col < node.display.col && (parent_node.display.row != node.display.row)) {
+            if (parent_node.display.row != node.display.row && parent_node.display.col != node.display.col) {
                 let connector = connect_elem.cloneNode()
                 connector.style.backgroundImage = "url('../media/atree/connect_angle.png')";
-                connector.classList.add("rotate-270");
                 connector.id = "r" + parent_node.display.row + "-c" + node.display.col + "-angle"
                 document.getElementById("atree-row-" + parent_node.display.row).children[node.display.col].appendChild(connector);
-                resolve_connector(document.getElementById("atree-row-" + parent_node.display.row).children[node.display.col], node);
+                if (parent_node.display.col > node.display.col) {
+                    connector.classList.add("rotate-180");
+                }
+                else {// if (parent_node.display.col < node.display.col && (parent_node.display.row != node.display.row)) {
+                    connector.classList.add("rotate-270");
+                }
+                resolve_connector(document.getElementById("atree-row-" + parent_node.display.row).children[node.display.col]);
             }
         }