update model

Small model update