Focus undotree when opened