Need to change tree data structure