add other models in api