minor chages to model