1 files changed, 3 insertions, 2 deletions
diff --git a/src/Q10/TrainTimeTable.java b/src/Q10/TrainTimeTable.java
index 6ef6abb..b79f10e 100644
--- a/src/Q10/TrainTimeTable.java
+++ b/src/Q10/TrainTimeTable.java
@@ -17,7 +17,7 @@ public class TrainTimeTable {
   public void delay(String city, int minutes) {
     System.out.println(String.format("Delay %s by %d minutes", city, minutes));
     Station station = stationIn(city);
-    station.delayBy(minutes);
+    if (station != null) station.delayBy(minutes);
   }
 
   public void displaySchedule() {
@@ -35,7 +35,8 @@ public class TrainTimeTable {
   }
 
   public Station stationIn(String city) {
-    for (Station station : this.schedule) if (station.getCity() == city) return station;
+    for (Station station : this.schedule)
+      if (station.getCity().toLowerCase().equals(city.toLowerCase())) return station;
     return null;
   }