Update classes for new layout.