added some model stuff