Make the train.Run func support custom http.ServeMux.