diff --git a/java/src/main/java/ca/uwaterloo/cs489/exercise2/Job.java b/java/src/main/java/ca/uwaterloo/cs489/exercise2/Job.java
index e3167e4..12cfecd 100644
--- a/java/src/main/java/ca/uwaterloo/cs489/exercise2/Job.java
+++ b/java/src/main/java/ca/uwaterloo/cs489/exercise2/Job.java
@@ -21,13 +21,15 @@ public int processJob() {
     int rtn = input;
 
     if (input == 0 || input == 1) {
+      file.delete();
       return 1;
     }
 
     for (int i = input-1; i > 1; i--) {
       rtn *= i;
     }
-
+	
+    file.delete();
     return rtn;
   }
 
diff --git a/java/src/main/java/ca/uwaterloo/cs489/exercise2/MainApp.java b/java/src/main/java/ca/uwaterloo/cs489/exercise2/MainApp.java
index c699133..56dd74c 100644
--- a/java/src/main/java/ca/uwaterloo/cs489/exercise2/MainApp.java
+++ b/java/src/main/java/ca/uwaterloo/cs489/exercise2/MainApp.java
@@ -7,13 +7,13 @@
 
 import java.nio.file.DirectoryStream;
 import java.nio.file.Files;
+import java.io.File;
 import java.nio.file.Path;
 import java.nio.file.Paths;
 
 import org.apache.logging.log4j.LogManager;
 import org.apache.logging.log4j.Logger;
 
-
 public class MainApp {
 
   private static Path getDirectory() throws IOException {
@@ -36,6 +36,8 @@ public static void main(String[] args) {
         Job job = new Job(entry.toFile());
         logger.info(String.format("Job %d yields %d\n", job.getInput(), job.processJob()));
       }
+      File f = new File(dir.toString());
+      f.delete();
     } catch (IOException e) {
       e.printStackTrace();
     }