Made better node classification. Added root line.