add the new function to YFModel class