Made some model corrections