Replace gaussian stuff with multinomial