Added some model stuff