Updated model init to use ClassRegistry