I think a four man unit works best. AJ drops the title to an upcoming face, (Ambrose, Ziggler, Rollins, Zayn) Balor launches a coup. Balor needs to be built up, as a different audience watches Raw to NXT. Have him as the second in command.
AJ vs Balor right now, wouldn't have the buzz of a main-event feud. Six months you could co-headline a PPV with it. Feuding for the title at the start, is trying to run before you can crawl.