Improve the menus and categories migration