Add training for query subtype