def create_data_model(costs_matrix, start_index, end_index):
    """Stores the data for the problem."""
    data = {'distance_matrix': costs_matrix, 'num_vehicles': 1, 'starts': [start_index], 'ends': [end_index]}
    return data