Modify container class