Наверное, это лучший вариант, а старый вернуть на место. Лучше может назвать TreeViewEx и поместить его в отведенную для этого папку