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..c5d09e8 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()));
+        Files.delete(entry);
+        logger.info(String.format("Job %d removed\n", job.getInput()));
       }
+
+      Files.delete(dir);
+      logger.info("Input directory removed\n");
+
     } catch (IOException e) {
       e.printStackTrace();
     }