enhance our model