Replaced the ITreeGridItem with a class