Simple decision tree library in Python