Refactor to use existing treenodeui rather than adding a new one (@danyaPostfactum)