diff --git a/src/main/java/duke/DukeDialogBox.java b/src/main/java/duke/DukeDialogBox.java
index bf08814d9d..a6289f18b8 100644
--- a/src/main/java/duke/DukeDialogBox.java
+++ b/src/main/java/duke/DukeDialogBox.java
@@ -1,5 +1,8 @@
 package duke;
 
+import java.io.IOException;
+import java.util.Collections;
+
 import javafx.collections.FXCollections;
 import javafx.collections.ObservableList;
 import javafx.fxml.FXML;
@@ -11,9 +14,6 @@
 import javafx.scene.image.ImageView;
 import javafx.scene.layout.HBox;
 
-import java.io.IOException;
-import java.util.Collections;
-
 public class DukeDialogBox extends HBox {
     @FXML
     private Label dialog;
diff --git a/src/main/java/duke/MainWindow.java b/src/main/java/duke/MainWindow.java
index 5428849572..d6a93b5f33 100644
--- a/src/main/java/duke/MainWindow.java
+++ b/src/main/java/duke/MainWindow.java
@@ -32,6 +32,9 @@ public class MainWindow extends AnchorPane {
     private Image userImage = new Image(this.getClass().getResourceAsStream("/images/DaUser.png"));
     private Image dukeImage = new Image(this.getClass().getResourceAsStream("/images/DaDuke.png"));
 
+    /**
+     * Initializes the GUI.
+     */
     @FXML
     public void initialize() {
         scrollPane.vvalueProperty().bind(dialogContainer.heightProperty());