diff --git a/src/main/java/seedu/duke/flashcard/FlashcardDirectory.java b/src/main/java/seedu/duke/flashcard/FlashcardDirectory.java
index c0aece5b31..b4d659a215 100644
--- a/src/main/java/seedu/duke/flashcard/FlashcardDirectory.java
+++ b/src/main/java/seedu/duke/flashcard/FlashcardDirectory.java
@@ -5,6 +5,8 @@
 /**
  * directory for flashcards
  * can list-up txt files in text folder
+ * In version 1, using only flashcard.txt
+ * In version 2, can select or create
  */
 
 public class FlashcardDirectory {
@@ -45,4 +47,23 @@ public void listFlashcardFiles(){
         }
     }
 
+    /**
+     * return default directory
+     * for version 1
+     * @return directory for flashcard txt file
+     */
+    public String defaultDirectory() {
+        return this.path + "/flashcard.txt";
+    }
+
+    /**
+     * return directory of flashcard txt file
+     * for version 2
+     * @param path
+     * @return
+     */
+    public String flashcardDirectory(String path) {
+        return this.path + path;
+    }
+
 }
diff --git a/src/main/java/seedu/duke/flashcard/FlashcardStorage.java b/src/main/java/seedu/duke/flashcard/FlashcardStorage.java
index 20b3a44a9b..2143b6a705 100644
--- a/src/main/java/seedu/duke/flashcard/FlashcardStorage.java
+++ b/src/main/java/seedu/duke/flashcard/FlashcardStorage.java
@@ -24,6 +24,12 @@ public FlashcardStorage(String path){
     }
 
 
+    /**
+     * load a flash card from certain format
+     * Tokens includes attributes of Flashcard
+     * @param tokens
+     * @return Flashcard object
+     */
     private Flashcard loadFlashcard(String[] tokens){
         String frontText = tokens[0].trim();
         String backText = tokens[1].trim();
@@ -56,6 +62,12 @@ private Flashcard loadFlashcard(String[] tokens){
         return flashcard;
     }
 
+    /**
+     * load list of flashcards
+     * from this.path
+     * @return list of Flashcards
+     * @throws FileNotFoundException
+     */
     public FlashcardList loadFlashcards() throws FileNotFoundException{
         FlashcardList flashcardList = new FlashcardList(new ArrayList<>());
         File f = new File (this.path);
@@ -84,4 +96,7 @@ public void saveFlashcards(ArrayList<Flashcard> flashcardList) {
             System.out.println("Failed to save.");
         }
     }
+
+
+
 }