Change *args to **kwargs for search_trains