delay all remaining stations and check for nulls

1 files changed, 4 insertions, 1 deletions

diff --git a/src/Q10/Station.java b/src/Q10/Station.java
index a2b2ae2..0aa35b8 100644
--- a/src/Q10/Station.java
+++ b/src/Q10/Station.java
@@ -61,7 +61,8 @@ public class Station {
     this.setDepartureDate(advanceDate(this.getDepartureDate(), minutes));
 
     Calendar calendar = Calendar.getInstance();
-    calendar.setTime(this.getDepartureDate());
+    calendar.setTime(
+        this.getDepartureDate() != null ? this.getDepartureDate() : this.getArrivalDate());
     calendar.setTimeZone(TimeZone.getTimeZone("UTC"));
     this.setDay(calendar.get(Calendar.DAY_OF_MONTH));
   }
@@ -75,6 +76,8 @@ public class Station {
   }
 
   private Date advanceDate(Date original, int minutes) {
+    if (original == null) return null;
+
     return new Date(original.getTime() + (minutes * 60000));
   }
 }