1 files changed, 3 insertions, 26 deletions

diff --git a/src/main/ui/BrowserCanvas.java b/src/main/ui/BrowserCanvas.java
index a58f6e1..43481ef 100644
--- a/src/main/ui/BrowserCanvas.java
+++ b/src/main/ui/BrowserCanvas.java
@@ -1,24 +1,17 @@
 package ui;
 
-import model.html.ElementNode;
-import model.html.TextNode;
-import model.html.Node;
-import model.layout.DocumentLayout;
-import model.layout.Layout;
+import model.html.*;
+import model.layout.*;
 
 import javax.swing.*;
 import java.awt.*;
 import java.util.*;
 
 public class BrowserCanvas extends JPanel {
-    private ArrayList<Node> html;
-    private DocumentLayout currentLayout;
+    private final DocumentLayout currentLayout;
 
-    // MODIFIES: this
-    // EFFECTS: constructs a BrowserCanvas object
     public BrowserCanvas(ArrayList<Node> html) {
         super();
-        this.html = html;
         this.currentLayout = Layout.constructTree(html);
         printTree(this.currentLayout.getChildren());
     }
@@ -30,7 +23,6 @@ public class BrowserCanvas extends JPanel {
         }
     }
 
-    // EFFECTS: paints a component onto a canvas
     @Override
     public void paintComponent(Graphics g) {
         super.paintComponent(g);
@@ -38,20 +30,6 @@ public class BrowserCanvas extends JPanel {
         renderHtml(this.currentLayout.getChildren(), g, location);
     }
 
-    // EFFECTS: naively renders our html file by printing text nodes
-    /*private void renderHtml(ArrayList<Node> html, Graphics g, Point location) {
-        for (Node node : html) {
-            if (node instanceof TextNode) {
-                if (node.getData().length() > 5) {
-                    g.drawString(node.getData(), location.x, location.y);
-                    location.translate(0, 20);
-                }
-            } else {
-                renderHtml(((ElementNode) node).getChildren(), g, location);
-            }
-        }
-    }*/
-
     private void renderHtml(ArrayList<Layout> tree, Graphics g, Point location) {
         for (Layout layout : tree) {
 //            System.out.println(layout.getLocation());
@@ -67,5 +45,4 @@ public class BrowserCanvas extends JPanel {
             }
         }
     }
-
 }