Slightly change the UI and add menu for categories