def add_line(self, line_number, station_name): if line_number in self.lines: self.lines[line_number].add_station(station_name) else: line = Line(line_number) line.add_station(station_name) self.lines[line_number] = line