Change to container on class for section