add multinomial regression code