Move tree right click menu into refed class