Add container class