Fixes and the start of unit tests for FeatureSplitter