diff --git a/src/se/edu/inclass/Main.java b/src/se/edu/inclass/Main.java index 3781393..938795c 100644 --- a/src/se/edu/inclass/Main.java +++ b/src/se/edu/inclass/Main.java @@ -21,6 +21,7 @@ public static void main(String[] args) { System.out.println("Total number of deadlines: " + countDeadlines(tasksData)); + printDeadlines(tasksData); } private static int countDeadlines(ArrayList tasksData) { @@ -40,10 +41,8 @@ public static void printData(ArrayList tasksData) { } public static void printDeadlines(ArrayList tasksData) { - for (Task t : tasksData) { - if (t instanceof Deadline) { - System.out.println(t); - } - } + tasksData.stream() + .filter((t) -> t instanceof Deadline) + .forEach(System.out::println); } }