Added model entity framework