Update menu code