Add loadTree method