add criterion parameter to decisiontree