Add missing container class