Improve ui, add callback for menu items