diff --git a/year4/semester2/CT414/assignments/assignment2/code/MapReduceFiles.java b/year4/semester2/CT414/assignments/assignment2/code/MapReduceFiles.java
index 6be28d36..72f43545 100644
--- a/year4/semester2/CT414/assignments/assignment2/code/MapReduceFiles.java
+++ b/year4/semester2/CT414/assignments/assignment2/code/MapReduceFiles.java
@@ -24,6 +24,11 @@ public class MapReduceFiles {
       input.put(args[0], readFile(args[0]));
       input.put(args[1], readFile(args[1]));
       input.put(args[2], readFile(args[2]));
+
+      for (String filename : args) {
+        input.put(filename, readFile(filename));
+      }
+
     }
     catch (IOException ex)
     {