Segment the stem and add light computation