Add leaf class.  Fill in training logic.