Almost complete decision tree
Small tweak