diff --git a/.gradle/6.7/executionHistory/executionHistory.bin b/.gradle/6.7/executionHistory/executionHistory.bin
index 6d4e076..ed7d052 100644
Binary files a/.gradle/6.7/executionHistory/executionHistory.bin and b/.gradle/6.7/executionHistory/executionHistory.bin differ
diff --git a/.gradle/6.7/executionHistory/executionHistory.lock b/.gradle/6.7/executionHistory/executionHistory.lock
index 6c7d418..910156d 100644
Binary files a/.gradle/6.7/executionHistory/executionHistory.lock and b/.gradle/6.7/executionHistory/executionHistory.lock differ
diff --git a/.gradle/6.7/fileHashes/fileHashes.bin b/.gradle/6.7/fileHashes/fileHashes.bin
index 8c5f67b..2365b93 100644
Binary files a/.gradle/6.7/fileHashes/fileHashes.bin and b/.gradle/6.7/fileHashes/fileHashes.bin differ
diff --git a/.gradle/6.7/fileHashes/fileHashes.lock b/.gradle/6.7/fileHashes/fileHashes.lock
index 00cd034..6279139 100644
Binary files a/.gradle/6.7/fileHashes/fileHashes.lock and b/.gradle/6.7/fileHashes/fileHashes.lock differ
diff --git a/.gradle/6.7/javaCompile/classAnalysis.bin b/.gradle/6.7/javaCompile/classAnalysis.bin
index efa3913..c0db3d7 100644
Binary files a/.gradle/6.7/javaCompile/classAnalysis.bin and b/.gradle/6.7/javaCompile/classAnalysis.bin differ
diff --git a/.gradle/6.7/javaCompile/javaCompile.lock b/.gradle/6.7/javaCompile/javaCompile.lock
index 837a255..3501837 100644
Binary files a/.gradle/6.7/javaCompile/javaCompile.lock and b/.gradle/6.7/javaCompile/javaCompile.lock differ
diff --git a/.gradle/buildOutputCleanup/buildOutputCleanup.lock b/.gradle/buildOutputCleanup/buildOutputCleanup.lock
index 5c783a8..bf5b26f 100644
Binary files a/.gradle/buildOutputCleanup/buildOutputCleanup.lock and b/.gradle/buildOutputCleanup/buildOutputCleanup.lock differ
diff --git a/.gradle/buildOutputCleanup/outputFiles.bin b/.gradle/buildOutputCleanup/outputFiles.bin
index 7fb98b9..de1ca2a 100644
Binary files a/.gradle/buildOutputCleanup/outputFiles.bin and b/.gradle/buildOutputCleanup/outputFiles.bin differ
diff --git a/.idea/gradle.xml b/.idea/gradle.xml
index 001104e..da925f2 100644
--- a/.idea/gradle.xml
+++ b/.idea/gradle.xml
@@ -8,7 +8,7 @@
         <option name="testRunner" value="GRADLE" />
         <option name="distributionType" value="DEFAULT_WRAPPED" />
         <option name="externalProjectPath" value="$PROJECT_DIR$" />
-        <option name="gradleJvm" value="adopt-openjdk-15" />
+        <option name="gradleJvm" value="openjdk-15" />
         <option name="modules">
           <set>
             <option value="$PROJECT_DIR$" />
diff --git a/build/classes/java/main/com/project/controller/Application.class b/build/classes/java/main/com/project/controller/Application.class
index 5fe26ca..ed1973a 100644
Binary files a/build/classes/java/main/com/project/controller/Application.class and b/build/classes/java/main/com/project/controller/Application.class differ
diff --git a/build/classes/java/main/com/project/controller/EntryPoint.class b/build/classes/java/main/com/project/controller/EntryPoint.class
index fd21f7e..273dd60 100644
Binary files a/build/classes/java/main/com/project/controller/EntryPoint.class and b/build/classes/java/main/com/project/controller/EntryPoint.class differ
diff --git a/build/classes/java/main/com/project/utility/GraphBenchmarkWorker.class b/build/classes/java/main/com/project/utility/GraphBenchmarkWorker.class
index 6898677..76d544f 100644
Binary files a/build/classes/java/main/com/project/utility/GraphBenchmarkWorker.class and b/build/classes/java/main/com/project/utility/GraphBenchmarkWorker.class differ
diff --git a/build/classes/java/main/com/project/utility/GraphGeneratorWorker.class b/build/classes/java/main/com/project/utility/GraphGeneratorWorker.class
index 8fe9e74..cf1a18a 100644
Binary files a/build/classes/java/main/com/project/utility/GraphGeneratorWorker.class and b/build/classes/java/main/com/project/utility/GraphGeneratorWorker.class differ
diff --git a/build/classes/java/main/com/project/utility/MinimumSpanningTreeKruskal.class b/build/classes/java/main/com/project/utility/MinimumSpanningTreeKruskal.class
index e77a778..898ce5f 100644
Binary files a/build/classes/java/main/com/project/utility/MinimumSpanningTreeKruskal.class and b/build/classes/java/main/com/project/utility/MinimumSpanningTreeKruskal.class differ
diff --git a/build/classes/java/main/com/project/utility/MinimumSpanningTreePrim.class b/build/classes/java/main/com/project/utility/MinimumSpanningTreePrim.class
index f3f29b6..2f569fe 100644
Binary files a/build/classes/java/main/com/project/utility/MinimumSpanningTreePrim.class and b/build/classes/java/main/com/project/utility/MinimumSpanningTreePrim.class differ
diff --git a/build/classes/java/main/com/project/utility/MyAlgorithm$algorithms.class b/build/classes/java/main/com/project/utility/MyAlgorithm$algorithms.class
index e67c4ac..5db6ebe 100644
Binary files a/build/classes/java/main/com/project/utility/MyAlgorithm$algorithms.class and b/build/classes/java/main/com/project/utility/MyAlgorithm$algorithms.class differ
diff --git a/build/classes/java/main/com/project/utility/MyAlgorithm.class b/build/classes/java/main/com/project/utility/MyAlgorithm.class
index ae8056a..f195250 100644
Binary files a/build/classes/java/main/com/project/utility/MyAlgorithm.class and b/build/classes/java/main/com/project/utility/MyAlgorithm.class differ
diff --git a/build/classes/java/main/com/project/view/MSTResultView.class b/build/classes/java/main/com/project/view/MSTResultView.class
index 3684d09..c2e911c 100644
Binary files a/build/classes/java/main/com/project/view/MSTResultView.class and b/build/classes/java/main/com/project/view/MSTResultView.class differ
diff --git a/build/classes/java/main/com/project/view/MainView.class b/build/classes/java/main/com/project/view/MainView.class
new file mode 100644
index 0000000..b2bbd6a
Binary files /dev/null and b/build/classes/java/main/com/project/view/MainView.class differ
diff --git a/build/resources/main/PC.png b/build/resources/main/PC.png
new file mode 100644
index 0000000..3db9b38
Binary files /dev/null and b/build/resources/main/PC.png differ
diff --git a/build/resources/main/Redpc.png b/build/resources/main/Redpc.png
new file mode 100644
index 0000000..9858baa
Binary files /dev/null and b/build/resources/main/Redpc.png differ
diff --git a/build/tmp/compileJava/source-classes-mapping.txt b/build/tmp/compileJava/source-classes-mapping.txt
index 74f0a44..de3b871 100644
--- a/build/tmp/compileJava/source-classes-mapping.txt
+++ b/build/tmp/compileJava/source-classes-mapping.txt
@@ -4,22 +4,24 @@ com/project/view/ChartResultView.java
  com.project.view.ChartResultView$dataCompareTypes
 com/project/model/Subset.java
  com.project.model.Subset
-com/project/utility/MinimumSpanningTreePrim.java
- com.project.utility.MinimumSpanningTreePrim
-com/project/utility/MinimumSpanningTreeKruskal.java
- com.project.utility.MinimumSpanningTreeKruskal
-com/project/view/MSTResultView.java
- com.project.view.MSTResultView
 com/project/utility/SetUtil.java
  com.project.utility.SetUtil
 com/project/utility/GraphGeneratorWorker.java
  com.project.utility.GraphGeneratorWorker
-com/project/controller/Application.java
- com.project.controller.Application
 com/project/utility/GraphBenchmarkWorker.java
  com.project.utility.GraphBenchmarkWorker
+com/project/utility/MinimumSpanningTreePrim.java
+ com.project.utility.MinimumSpanningTreePrim
+com/project/utility/MinimumSpanningTreeKruskal.java
+ com.project.utility.MinimumSpanningTreeKruskal
+com/project/view/MSTResultView.java
+ com.project.view.MSTResultView
 com/project/utility/MyAlgorithm.java
  com.project.utility.MyAlgorithm
  com.project.utility.MyAlgorithm$algorithms
+com/project/view/MainView.java
+ com.project.view.MainView
+com/project/controller/Application.java
+ com.project.controller.Application
 com/project/controller/EntryPoint.java
  com.project.controller.EntryPoint
diff --git a/src/main/java/com/project/controller/Application.java b/src/main/java/com/project/controller/Application.java
index 903a118..fd74a30 100644
--- a/src/main/java/com/project/controller/Application.java
+++ b/src/main/java/com/project/controller/Application.java
@@ -1,220 +1,101 @@
 package com.project.controller;
 
 import com.project.utility.GraphBenchmarkWorker;
-import com.project.utility.MyAlgorithm;
-import com.project.view.ChartResultView;
-import com.project.view.MSTResultView;
+import com.project.view.MainView;
 import org.graphstream.graph.Graph;
 
-import javax.swing.*;
-import java.awt.*;
 import java.util.Collection;
 import java.util.List;
 import java.util.concurrent.ExecutionException;
 
-public class Application extends JFrame implements Runnable {
-    private MSTResultView kruskalResultView;
-    private MSTResultView primResultView;
-    private ChartResultView chartResultView;
-
-    private JTabbedPane resultTabbedPane;
-    private JPanel kruskalTab;
-    private JPanel primTab;
-    private JPanel kruskalGraphPanel;
-    private JPanel primGraphPanel;
-
-    private JTextField numberOfNodesPerStepTextField;
-    private JTextField numberOfStepsTextField;
-    private JTextField iterationsPerStepTextField;
-
-    private JButton generateButton;
-    private JButton clearButton;
-    private JButton runButton;
-
-    private boolean built;
+public class Application {
     private int numberOfVertexPerStep, maxNumberOfIteration, iterationPerStep;
 
-    @Override
-    public void run() {
-        System.setProperty("org.graphstream.ui", "swing");
-        this.setTitle("Minimum Spanning Network");
-        this.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
-        this.setPreferredSize(new Dimension(1280, 720));
-
-        initLeftPane(this);
-
-        resultTabbedPane = new JTabbedPane();
-
-        this.getContentPane().add(resultTabbedPane);
-
-        this.setSize(this.getPreferredSize());
-        this.pack();
-        this.setVisible(true);
-    }
-
-    private void initLeftPane(Container container) {
-        JPanel topPanel = new JPanel();
+    public void start() {
+        MainView mainView = new MainView();
 
-        FlowLayout flowLayout = new FlowLayout();
-        flowLayout.setHgap(5);
+        mainView.display();
 
-        topPanel.setLayout(flowLayout);
-        topPanel.setBorder(BorderFactory.createLineBorder(Color.gray, 2));
+        mainView.getGenerateButton().addActionListener(event -> {
+            mainView.getGenerateButton().setEnabled(false);
+            mainView.getClearButton().setEnabled(true);
+            mainView.getRunButton().setEnabled(true);
 
-        topPanel.add(new JLabel("Nodes per step ="));
-        numberOfNodesPerStepTextField = new JTextField();
-        numberOfNodesPerStepTextField.setPreferredSize(new Dimension(70, 20));
-        topPanel.add(numberOfNodesPerStepTextField);
-
-        topPanel.add(new JLabel("Number of steps ="));
-        numberOfStepsTextField = new JTextField();
-        numberOfStepsTextField.setPreferredSize(new Dimension(70, 20));
-        topPanel.add(numberOfStepsTextField);
-
-        topPanel.add(new JLabel("Iteration per step ="));
-        iterationsPerStepTextField = new JTextField();
-        iterationsPerStepTextField.setPreferredSize(new Dimension(70, 20));
-        topPanel.add(iterationsPerStepTextField);
-
-        initButtons();
-
-        topPanel.add(generateButton);
-        topPanel.add(clearButton);
-        topPanel.add(runButton);
-
-        container.add(topPanel, BorderLayout.NORTH);
-    }
-
-    private void initGraphPanel() {
-        kruskalGraphPanel = new JPanel();
-        kruskalGraphPanel.setLayout(new GridLayout());
-        kruskalGraphPanel.setBorder(BorderFactory.createLineBorder(Color.gray, 2));
-
-        primGraphPanel = new JPanel();
-        primGraphPanel.setLayout(new GridLayout());
-        primGraphPanel.setBorder(BorderFactory.createLineBorder(Color.gray, 2));
-    }
+            mainView.initGraphPanel();
 
-    private void initButtons() {
-        Dimension dimension = new Dimension(100, 20);
+            this.numberOfVertexPerStep = Integer.parseInt(mainView.getNumberOfNodesPerStepTextField().getText());
+            this.maxNumberOfIteration = Integer.parseInt(mainView.getNumberOfStepsTextField().getText()) * numberOfVertexPerStep;
+            this.iterationPerStep = Integer.parseInt(mainView.getIterationsPerStepTextField().getText());
 
-        generateButton = new JButton("Generate");
-        generateButton.setPreferredSize(dimension);
+            mainView.getKruskalResultView().init(numberOfVertexPerStep, maxNumberOfIteration, iterationPerStep);
+            mainView.getPrimResultView().init(numberOfVertexPerStep, maxNumberOfIteration, iterationPerStep);
 
-        generateButton.addActionListener(event -> {
-            if (!built) {
-                generateButton.setEnabled(false);
-                clearButton.setEnabled(true);
-                runButton.setEnabled(true);
+            mainView.getKruskalResultView().execute();
 
-                initGraphPanel();
-
-                kruskalTab = new JPanel();
-                primTab = new JPanel();
-
-                kruskalTab.setLayout(new GridLayout());
-                primTab.setLayout(new GridLayout());
-
-                resultTabbedPane.addTab("Kruskal", kruskalTab);
-                resultTabbedPane.addTab("Prim", primTab);
-
-                kruskalTab.add(kruskalGraphPanel);
-                primTab.add(primGraphPanel);
-
-                kruskalResultView = new MSTResultView(kruskalGraphPanel, MyAlgorithm.algorithms.KRUSKAL);
-                primResultView = new MSTResultView(primGraphPanel, MyAlgorithm.algorithms.PRIM);
-                chartResultView = new ChartResultView(resultTabbedPane);
-
-                kruskalGraphPanel.add(kruskalResultView.getTabbedPane());
-                primGraphPanel.add(primResultView.getTabbedPane());
-
-                this.numberOfVertexPerStep = Integer.parseInt(numberOfNodesPerStepTextField.getText());
-                this.maxNumberOfIteration = Integer.parseInt(numberOfStepsTextField.getText()) * numberOfVertexPerStep;
-                this.iterationPerStep = Integer.parseInt(iterationsPerStepTextField.getText());
-
-                kruskalResultView.init(numberOfVertexPerStep, maxNumberOfIteration, iterationPerStep);
-                primResultView.init(numberOfVertexPerStep, maxNumberOfIteration, iterationPerStep);
-
-                kruskalResultView.execute();
+            try {
+                mainView.getKruskalResultView().get();
+            } catch (InterruptedException | ExecutionException interruptedException) {
+                interruptedException.printStackTrace();
+            }
 
+            if (mainView.getKruskalResultView().isDone()) {
+                mainView.getPrimResultView().execute();
+                System.out.println("Kruskal done!");
                 try {
-                    kruskalResultView.get();
+                    mainView.getPrimResultView().get();
                 } catch (InterruptedException | ExecutionException interruptedException) {
                     interruptedException.printStackTrace();
                 }
-
-                if (kruskalResultView.isDone()) {
-                    primResultView.execute();
-                    System.out.println("Kruskal done!");
-                    try {
-                        primResultView.get();
-                    } catch (InterruptedException | ExecutionException interruptedException) {
-                        interruptedException.printStackTrace();
-                    }
-                }
-
-                built = true;
             }
         });
 
-        clearButton = new JButton("Clear");
-        clearButton.setPreferredSize(dimension);
+        mainView.getClearButton().addActionListener(event -> {
+            mainView.getGenerateButton().setEnabled(true);
+            mainView.getClearButton().setEnabled(false);
+            mainView.getRunButton().setEnabled(false);
 
-        clearButton.addActionListener(event -> {
-            generateButton.setEnabled(true);
-            clearButton.setEnabled(false);
-            runButton.setEnabled(false);
-
-            kruskalResultView.getGraphLists().stream()
+            mainView.getKruskalResultView().getGraphLists().stream()
                     .flatMap(Collection::parallelStream)
                     .forEach(Graph::clear);
 
-            primResultView.getGraphLists().stream()
+            mainView.getPrimResultView().getGraphLists().stream()
                     .flatMap(Collection::parallelStream)
                     .forEach(Graph::clear);
 
-            kruskalResultView.getSwingViewer().getGraphicGraph().clear();
-            primResultView.getSwingViewer().getGraphicGraph().clear();
-
-            resultTabbedPane.removeAll();
+            mainView.getKruskalResultView().getSwingViewer().getGraphicGraph().clear();
+            mainView.getPrimResultView().getSwingViewer().getGraphicGraph().clear();
 
-            built = false;
+            mainView.getResultTabbedPane().removeAll();
         });
 
-        runButton = new JButton("Run");
-        runButton.setPreferredSize(dimension);
-
-        runButton.addActionListener(event -> {
-            generateButton.setEnabled(false);
-            clearButton.setEnabled(true);
-            runButton.setEnabled(false);
+        mainView.getRunButton().addActionListener(event -> {
+            mainView.getGenerateButton().setEnabled(false);
+            mainView.getClearButton().setEnabled(true);
+            mainView.getRunButton().setEnabled(false);
 
             try {
                 GraphBenchmarkWorker graphBenchmarkWorker = new GraphBenchmarkWorker();
 
-                graphBenchmarkWorker.init(kruskalResultView.getGraphLists(), kruskalResultView.getAlgorithm());
+                graphBenchmarkWorker.init(mainView.getKruskalResultView().getGraphLists(), mainView.getKruskalResultView().getAlgorithm());
                 graphBenchmarkWorker.compute();
 
                 List<List<Double>> kruskalResults = graphBenchmarkWorker.getBenchmarkLists();
 
-                graphBenchmarkWorker.init(primResultView.getGraphLists(), primResultView.getAlgorithm());
+                graphBenchmarkWorker.init(mainView.getPrimResultView().getGraphLists(), mainView.getPrimResultView().getAlgorithm());
                 graphBenchmarkWorker.compute();
 
                 List<List<Double>> primResults = graphBenchmarkWorker.getBenchmarkLists();
 
-                chartResultView.draw(
+                mainView.getChartResultView().draw(
                         kruskalResults,
                         primResults,
                         numberOfVertexPerStep);
 
-                this.validate();
-                this.repaint();
+                mainView.validate();
+                mainView.repaint();
             } catch (Exception e) {
                 e.printStackTrace();
             }
         });
-
-        clearButton.setEnabled(false);
-        runButton.setEnabled(false);
     }
 }
diff --git a/src/main/java/com/project/controller/EntryPoint.java b/src/main/java/com/project/controller/EntryPoint.java
index 63ce4a4..fc30e5b 100644
--- a/src/main/java/com/project/controller/EntryPoint.java
+++ b/src/main/java/com/project/controller/EntryPoint.java
@@ -4,6 +4,6 @@
 
 public class EntryPoint {
     public static void main(String[] args) {
-        SwingUtilities.invokeLater(new Application());
+        SwingUtilities.invokeLater(new Application()::start);
     }
 }
diff --git a/src/main/java/com/project/utility/GraphBenchmarkWorker.java b/src/main/java/com/project/utility/GraphBenchmarkWorker.java
index 2147669..ec58909 100644
--- a/src/main/java/com/project/utility/GraphBenchmarkWorker.java
+++ b/src/main/java/com/project/utility/GraphBenchmarkWorker.java
@@ -1,8 +1,8 @@
 package com.project.utility;
 
-import org.graphstream.graph.Edge;
 import org.graphstream.graph.Graph;
 
+import java.util.ArrayList;
 import java.util.LinkedList;
 import java.util.List;
 
@@ -24,48 +24,43 @@ public List<List<Double>> getBenchmarkLists() {
     }
 
     public void compute() {
-        MyAlgorithm myAlgorithm = null;
-
-        for (int i = 0; i < 1000; i++) {
-            myAlgorithm = (algorithms == MyAlgorithm.algorithms.KRUSKAL)
-                    ? new MinimumSpanningTreeKruskal()
-                    : new MinimumSpanningTreePrim();
-        }
+        List<Double> runTimes = new LinkedList<>();
+        MyAlgorithm myAlgorithm;
 
         for (List<Graph> graphList : graphLists) {
-            List<Double> runTimes = new LinkedList<>();
-
-            for (Graph graph : graphList) {
-                List<Double> temp = new LinkedList<>();
 
-                for (int i = 0; i < 130; i++) {
+            // Warm-up
+            for (int i = 0; i < 1000; i++) {
+                for (Graph graph : graphList) {
                     myAlgorithm = setup(algorithms);
                     myAlgorithm.init(graph);
-
-                    double startTime = System.nanoTime();
                     myAlgorithm.compute();
-                    double endTime = System.nanoTime();
-
-                    temp.add((endTime - startTime));
                 }
+            }
+
+            myAlgorithm = setup(algorithms);
+
+            // Real test
+            for (Graph graph : graphList) {
+                myAlgorithm.init(graph);
+
+                double startTime = System.nanoTime();
+                myAlgorithm.compute();
+                double endTime = System.nanoTime();
 
-                double value = temp.stream().min(Double::compare).orElseThrow();
+                runTimes.add((endTime - startTime) /1000000);
 
-                runTimes.add(value/1000000);
+                myAlgorithm.getMstResult()
+                        .forEach(edge -> edge.setAttribute("ui.style", "fill-color: blue; size: 5px;"));
 
-                List<Edge> mstArrayList = myAlgorithm.getMstResult();
-                mstArrayList.forEach(edge -> {
-                    edge.setAttribute("ui.style", "fill-color: blue; size: 5px;");
-                    edge.getNode0().setAttribute("ui.style", "fill-image: url('src/main/resources/PC.png');");
-                    edge.getNode1().setAttribute("ui.style", "fill-image: url('src/main/resources/PC.png');");
-                });
                 myAlgorithm.getSuperComputers().setAttribute("ui.style", "fill-color: red; size: 4px;");
-                myAlgorithm.getSuperComputers().getNode0().setAttribute("ui.style","fill-image: url('src/main/resources/Redpc.png'); ");
-                myAlgorithm.getSuperComputers().getNode1().setAttribute("ui.style","fill-image: url('src/main/resources/Redpc.png'); ");
+                myAlgorithm.getSuperComputers().getNode0().setAttribute("ui.class", "superComputer");
+                myAlgorithm.getSuperComputers().getNode1().setAttribute("ui.class", "superComputer");
                 myAlgorithm.getSuperComputers().setAttribute("weight", myAlgorithm.getOriginalWeight());
             }
 
-            benchmarkLists.add(runTimes);
+            benchmarkLists.add(new ArrayList<>(runTimes));
+            runTimes.clear();
         }
     }
 
diff --git a/src/main/java/com/project/utility/GraphGeneratorWorker.java b/src/main/java/com/project/utility/GraphGeneratorWorker.java
index 80dbbda..bc2fdfa 100644
--- a/src/main/java/com/project/utility/GraphGeneratorWorker.java
+++ b/src/main/java/com/project/utility/GraphGeneratorWorker.java
@@ -70,7 +70,8 @@ public void compute() {
                         "text-background-mode: rounded-box; " +
                         "text-background-color: #222C; " +
                         "text-padding: 1px; " +
-                        "text-offset: 0px, 2px;}");
+                        "text-offset: 0px, 2px;}" +
+                        "node.superComputer {fill-image: url('src/main/resources/Redpc.png');}");
 
         graph.setAttribute("ui.quality");
         graph.setAttribute("ui.antialias");
diff --git a/src/main/java/com/project/utility/MinimumSpanningTreeKruskal.java b/src/main/java/com/project/utility/MinimumSpanningTreeKruskal.java
index 727ac73..0659806 100644
--- a/src/main/java/com/project/utility/MinimumSpanningTreeKruskal.java
+++ b/src/main/java/com/project/utility/MinimumSpanningTreeKruskal.java
@@ -2,40 +2,10 @@
 
 import com.project.model.Subset;
 import org.graphstream.graph.Edge;
-import org.graphstream.graph.Graph;
 
 import java.util.*;
-import java.util.stream.Collectors;
 
 public class MinimumSpanningTreeKruskal extends MyAlgorithm {
-    private List<Edge> edgeList;
-
-    @Override
-    public void init(Graph graph) {
-        super.graph = graph;
-
-        Random random = new Random();
-        mstResult = new LinkedList<>();
-
-        edgeList = graph.edges().collect(Collectors.toList());
-
-        int minWeight = 1;
-        int maxWeight = 500;
-        for (Edge edge : edgeList) {
-            int weight = minWeight + random.nextInt(maxWeight);
-            edge.setAttribute("weight", weight);
-            edge.setAttribute("ui.label", weight);
-            edge.setAttribute("ui.style", "fill-color: gray;");
-        }
-
-        int randomEdge = random.nextInt(edgeList.size());
-        superComputers = edgeList.get(randomEdge);
-
-        originalWeight = (int) superComputers.getAttribute("weight");
-        superComputers.setAttribute("weight", 0);
-        superComputers.getNode0().setAttribute("ui.style", "fill-image: url('src/main/resources/Redpc.png'); ");
-        superComputers.getNode1().setAttribute("ui.style", "fill-image: url('src/main/resources/Redpc.png'); ");
-    }
 
     @Override
     public void compute() {
diff --git a/src/main/java/com/project/utility/MinimumSpanningTreePrim.java b/src/main/java/com/project/utility/MinimumSpanningTreePrim.java
index 64d4502..5b8c7e0 100644
--- a/src/main/java/com/project/utility/MinimumSpanningTreePrim.java
+++ b/src/main/java/com/project/utility/MinimumSpanningTreePrim.java
@@ -1,42 +1,12 @@
 package com.project.utility;
 
-import org.graphstream.graph.Edge;
-import org.graphstream.graph.Graph;
 import org.jheaps.AddressableHeap;
 import org.jheaps.tree.FibonacciHeap;
 
 import java.util.*;
-import java.util.stream.Collectors;
 
 public class MinimumSpanningTreePrim extends MyAlgorithm {
 
-    @Override
-    public void init(Graph graph) {
-        super.graph = graph;
-
-        Random random = new Random();
-        mstResult = new LinkedList<>();
-
-        List<Edge> edgeList = graph.edges().collect(Collectors.toList());
-
-        int minWeight = 1;
-        int maxWeight = 500;
-        for (Edge edge : edgeList) {
-            int weight = minWeight + random.nextInt(maxWeight);
-            edge.setAttribute("weight", weight);
-            edge.setAttribute("ui.label", weight);
-            edge.setAttribute("ui.style", "fill-color: gray;");
-        }
-
-        int randomEdge = random.nextInt(edgeList.size());
-        superComputers = edgeList.get(randomEdge);
-
-        originalWeight = (int) superComputers.getAttribute("weight");
-        superComputers.setAttribute("weight", 0);
-        superComputers.getNode0().setAttribute("ui.style", "fill-image: url('src/main/resources/Redpc.png'); ");
-        superComputers.getNode1().setAttribute("ui.style", "fill-image: url('src/main/resources/Redpc.png'); ");
-    }
-
     @Override
     public void compute() throws IndexOutOfBoundsException {
         boolean[] mstIncluded = new boolean[graph.getNodeCount()];
diff --git a/src/main/java/com/project/utility/MyAlgorithm.java b/src/main/java/com/project/utility/MyAlgorithm.java
index 1b75796..ed2218d 100644
--- a/src/main/java/com/project/utility/MyAlgorithm.java
+++ b/src/main/java/com/project/utility/MyAlgorithm.java
@@ -4,7 +4,10 @@
 import org.graphstream.graph.Edge;
 import org.graphstream.graph.Graph;
 
+import java.util.LinkedList;
 import java.util.List;
+import java.util.Random;
+import java.util.stream.Collectors;
 
 public abstract class MyAlgorithm implements Algorithm {
     public enum algorithms {
@@ -14,6 +17,7 @@ public enum algorithms {
     protected Graph graph;
     protected Edge superComputers;
     protected List<Edge> mstResult;
+    protected List<Edge> edgeList;
     protected int originalWeight;
 
     public Edge getSuperComputers() {
@@ -31,5 +35,25 @@ public int getOriginalWeight() {
     @Override
     public void init(Graph graph) {
         this.graph = graph;
+
+        Random random = new Random();
+        mstResult = new LinkedList<>();
+
+        edgeList = graph.edges().collect(Collectors.toList());
+
+        int minWeight = 1;
+        int maxWeight = 500;
+        for (Edge edge : edgeList) {
+            int weight = minWeight + random.nextInt(maxWeight);
+            edge.setAttribute("weight", weight);
+            edge.setAttribute("ui.label", weight);
+            edge.setAttribute("ui.style", "fill-color: gray;");
+        }
+
+        int randomEdge = random.nextInt(edgeList.size());
+        superComputers = edgeList.get(randomEdge);
+
+        originalWeight = (int) superComputers.getAttribute("weight");
+        superComputers.setAttribute("weight", 0);
     }
 }
diff --git a/src/main/java/com/project/view/MSTResultView.java b/src/main/java/com/project/view/MSTResultView.java
index 3d42e1d..03b336d 100644
--- a/src/main/java/com/project/view/MSTResultView.java
+++ b/src/main/java/com/project/view/MSTResultView.java
@@ -162,24 +162,17 @@ private void changeGraph(Graph graph) {
 
         swingViewer.getGraphicGraph().clear();
 
-        swingViewer.getGraphicGraph().setAttribute("ui.stylesheet",
-                "node {fill-mode:  image-scaled;" +
-                        "size: 32px;" +
-                        "fill-image: url('src/main/resources/PC.png'); shape: box; }"+
-                        "edge {fill-mode: dyn-plain;" +
-                        "text-alignment: under; " +
-                        "text-color: white; " +
-                        "text-style: bold; " +
-                        "text-background-mode: rounded-box; " +
-                        "text-background-color: #222C; " +
-                        "text-padding: 1px; " +
-                        "text-offset: 0px, 2px;}");
+        swingViewer.getGraphicGraph().setAttribute("ui.stylesheet", graph.getAttribute("ui.stylesheet"));
 
         try {
             graph.nodes().forEach(node -> {
                 double[] xyz = Toolkit.nodePosition(node);
                 Node changedNode = swingViewer.getGraphicGraph().addNode(node.getId());
-                changedNode.setAttribute("ui.style", node.getAttribute("ui.style"));
+
+                if (node.hasAttribute("ui.class")) {
+                    changedNode.setAttribute("ui.class", "superComputer");
+                }
+
                 changedNode.setAttribute("xyz", xyz[0], xyz[1], xyz[2]);
             });
 
@@ -187,7 +180,11 @@ private void changeGraph(Graph graph) {
                 Edge changedEdge = swingViewer
                         .getGraphicGraph()
                         .addEdge(edge.getId(), edge.getNode0().getId(), edge.getNode1().getId());
-                changedEdge.setAttribute("ui.style", edge.getAttribute("ui.style"));
+
+                if (edge.hasAttribute("ui.style")) {
+                    changedEdge.setAttribute("ui.style", edge.getAttribute("ui.style"));
+                }
+
                 changedEdge.setAttribute("ui.label", edge.getAttribute("ui.label"));
             });
         } catch (IndexOutOfBoundsException | IdAlreadyInUseException | EdgeRejectedException | ElementNotFoundException e) {
diff --git a/src/main/java/com/project/view/MainView.java b/src/main/java/com/project/view/MainView.java
new file mode 100644
index 0000000..c05fa4c
--- /dev/null
+++ b/src/main/java/com/project/view/MainView.java
@@ -0,0 +1,158 @@
+package com.project.view;
+
+import com.project.utility.MyAlgorithm;
+
+import javax.swing.*;
+import java.awt.*;
+
+public class MainView extends JFrame {
+    private MSTResultView kruskalResultView;
+    private MSTResultView primResultView;
+    private ChartResultView chartResultView;
+
+    private JTabbedPane resultTabbedPane;
+
+    private JTextField numberOfNodesPerStepTextField;
+    private JTextField numberOfStepsTextField;
+    private JTextField iterationsPerStepTextField;
+
+    private JButton generateButton;
+    private JButton clearButton;
+    private JButton runButton;
+
+    public void display() {
+        System.setProperty("org.graphstream.ui", "swing");
+
+        this.setTitle("Minimum Spanning Network");
+        this.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
+        this.setPreferredSize(new Dimension(1280, 720));
+
+        initLeftPane(this);
+
+        resultTabbedPane = new JTabbedPane();
+
+        this.getContentPane().add(resultTabbedPane);
+
+        clearButton.setEnabled(false);
+        runButton.setEnabled(false);
+
+        this.setSize(this.getPreferredSize());
+        this.pack();
+        this.setVisible(true);
+    }
+
+    public ChartResultView getChartResultView() {
+        return chartResultView;
+    }
+
+    public JTabbedPane getResultTabbedPane() {
+        return resultTabbedPane;
+    }
+
+    public MSTResultView getKruskalResultView() {
+        return kruskalResultView;
+    }
+
+    public MSTResultView getPrimResultView() {
+        return primResultView;
+    }
+
+    public JTextField getNumberOfNodesPerStepTextField() {
+        return numberOfNodesPerStepTextField;
+    }
+
+    public JTextField getNumberOfStepsTextField() {
+        return numberOfStepsTextField;
+    }
+
+    public JTextField getIterationsPerStepTextField() {
+        return iterationsPerStepTextField;
+    }
+
+    public JButton getGenerateButton() {
+        return generateButton;
+    }
+
+    public JButton getClearButton() {
+        return clearButton;
+    }
+
+    public JButton getRunButton() {
+        return runButton;
+    }
+
+    public void initGraphPanel() {
+        JPanel kruskalGraphPanel = new JPanel();
+        kruskalGraphPanel.setLayout(new GridLayout());
+        kruskalGraphPanel.setBorder(BorderFactory.createLineBorder(Color.gray, 2));
+
+        JPanel primGraphPanel = new JPanel();
+        primGraphPanel.setLayout(new GridLayout());
+        primGraphPanel.setBorder(BorderFactory.createLineBorder(Color.gray, 2));
+
+        JPanel kruskalTab = new JPanel();
+        JPanel primTab = new JPanel();
+
+        kruskalTab.setLayout(new GridLayout());
+        primTab.setLayout(new GridLayout());
+
+        resultTabbedPane.addTab("Kruskal", kruskalTab);
+        resultTabbedPane.addTab("Prim", primTab);
+
+        kruskalTab.add(kruskalGraphPanel);
+        primTab.add(primGraphPanel);
+
+        kruskalResultView = new MSTResultView(kruskalGraphPanel, MyAlgorithm.algorithms.KRUSKAL);
+        primResultView = new MSTResultView(primGraphPanel, MyAlgorithm.algorithms.PRIM);
+        chartResultView = new ChartResultView(resultTabbedPane);
+
+        kruskalGraphPanel.add(kruskalResultView.getTabbedPane());
+        primGraphPanel.add(primResultView.getTabbedPane());
+    }
+
+    private void initLeftPane(Container container) {
+        JPanel topPanel = new JPanel();
+
+        FlowLayout flowLayout = new FlowLayout();
+        flowLayout.setHgap(5);
+
+        topPanel.setLayout(flowLayout);
+        topPanel.setBorder(BorderFactory.createLineBorder(Color.gray, 2));
+
+        topPanel.add(new JLabel("Nodes per step ="));
+        numberOfNodesPerStepTextField = new JTextField();
+        numberOfNodesPerStepTextField.setPreferredSize(new Dimension(70, 20));
+        topPanel.add(numberOfNodesPerStepTextField);
+
+        topPanel.add(new JLabel("Number of steps ="));
+        numberOfStepsTextField = new JTextField();
+        numberOfStepsTextField.setPreferredSize(new Dimension(70, 20));
+        topPanel.add(numberOfStepsTextField);
+
+        topPanel.add(new JLabel("Iteration per step ="));
+        iterationsPerStepTextField = new JTextField();
+        iterationsPerStepTextField.setPreferredSize(new Dimension(70, 20));
+        topPanel.add(iterationsPerStepTextField);
+
+        initButtons();
+
+        topPanel.add(generateButton);
+        topPanel.add(clearButton);
+        topPanel.add(runButton);
+
+        container.add(topPanel, BorderLayout.NORTH);
+    }
+
+    private void initButtons() {
+        Dimension dimension = new Dimension(100, 20);
+
+        generateButton = new JButton("Generate");
+        generateButton.setPreferredSize(dimension);
+
+        clearButton = new JButton("Clear");
+        clearButton.setPreferredSize(dimension);
+
+        runButton = new JButton("Run");
+        runButton.setPreferredSize(dimension);
+    }
+}