Adding minor changes to the multi-nomial classifier