input: Write a Python function to predict the class of a new data point using a trained DecisionTreeClassifier model.
  assert predict_class(model, np.array([[2, 1, 1]])) == 1
  assert predict_class(model, np.array([[0, 2, 0]])) == 0
  assert predict_class(model, np.array([[1, 0, 1]])) == 1
output: def predict_class(model, data_point):
      return model.predict(data_point)[0]