Add missing classes based on side