added two new models
minor: fixed bug