fixing model
updated data model