Add back categories for better spacing