No More Softmaxing The Classes Of Drugs