@azasypkin we should do the same here, I guess?

Then maybe we need a parent.dataset.draftId and parent.dataset.threadId to make it clear that we have drafts and threads showing in the list. Then maybe it's too big for this patch and we could do all this (replacing the classes by datasets in the if conditions here) in a follow up? What do you think?