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..77a5901 100644
--- a/java/src/main/java/ca/uwaterloo/cs489/exercise2/MainApp.java
+++ b/java/src/main/java/ca/uwaterloo/cs489/exercise2/MainApp.java
@@ -35,7 +35,13 @@ public static void main(String[] args) {
       for (Path entry : ds) {
         Job job = new Job(entry.toFile());
         logger.info(String.format("Job %d yields %d\n", job.getInput(), job.processJob()));
+        File file = new File(entry.toUri());
+        file.delete();
+        logger.info("Deleting {}\n", entry.toUri());
       }
+      File directory = new File(dir.toUri());
+      logger.info("Deleting directory {}\n", dir.toUri());
+      directory.delete();
     } catch (IOException e) {
       e.printStackTrace();
     }