[Problem & Solution]: Decision Trees code