Made initializing model a little bit cleaner