Context menus now work, adding support for tree editor