Added extension capability, for robust split (and maybe NLP + stochastic too)

Yay for merge conflicts :+1: