Added code to generate decision stumps