updated model, copied code from sous