add entropy/information gain function