def search_trains(args: dict) -> dict: raise NotImplementedError("TODO implement :)") def book_trains(args: dict) -> dict: raise NotImplementedError("TODO implement :)")