Add support for multi classification and use results to narrow down subtype