Add test before implementing Gini properly