improve model a bit