diff --git a/src/main/java/Controller.java b/src/main/java/Controller.java
index a905ef4..c81fe60 100644
--- a/src/main/java/Controller.java
+++ b/src/main/java/Controller.java
@@ -19,17 +19,10 @@ public class Controller{
 
     @FXML
     private void initialize() {
-//        createTree();
-    }
 
-    // Create the directory Tree
-    private void createTree() {
-        TreeItem<String> root = new TreeItem<String>("Root");
-        TreeItem<String> itemChild = new TreeItem<String>("Child");
-        itemChild.setExpanded(false);
-        root.getChildren().add(itemChild);
-        dirTree.setRoot(root);
     }
 
+
+
 }
 
diff --git a/src/main/java/Main.java b/src/main/java/Main.java
index 82c258a..fb58fa5 100644
--- a/src/main/java/Main.java
+++ b/src/main/java/Main.java
@@ -5,9 +5,12 @@
 import javafx.geometry.Rectangle2D;
 import javafx.scene.Parent;
 import javafx.scene.Scene;
+import javafx.scene.control.TreeView;
 import javafx.stage.Screen;
 import javafx.stage.Stage;
 
+import java.io.File;
+
 public class Main extends Application {
 
     @Override
@@ -24,12 +27,26 @@ public void handle(ActionEvent event) {
         });*/
         Parent root = FXMLLoader.load(getClass().getResource("/main/resources/StartPage.fxml"));
         Rectangle2D screenBounds = Screen.getPrimary().getVisualBounds();
+
         primaryStage.setTitle("Iron-gate!");
-//        primaryStage.setScene(new Scene(root, screenBounds.getWidth(), screenBounds.getHeight())); // a scene is created here
         primaryStage.setScene(new Scene(root, 990, 785));
+
         Scene scene = primaryStage.getScene(); // we get the scene from above
         scene.getStylesheets().clear(); // clear any styles
-//        scene.getStylesheets().add("/main/resources/mainStyle.css"); // absolute path
+        scene.getStylesheets().add("/main/resources/mainStyle.css"); // absolute path
+
+        try { //try to set the tree view in the FolderViewManager class
+
+            FolderViewManager.treeView = (TreeView<String>) scene.lookup("#folder-view-pane");
+            File homeDir = new File(System.getProperty("user.home"));
+
+            FolderViewManager.setRootDirectory(homeDir);
+            System.out.println(homeDir.getName() + " set as root directory.");
+
+        } catch (Exception e) {
+            e.printStackTrace();
+        }
+
         primaryStage.show();
     }
     public static void main(String[] args) {