Model improvment