Hello. Thank you for your wonderful research. I want to extend your methodology to multi-class classification. Is this possible? If possible, could you give me some implementation tips?