Formatting of Figures 2 and 3 / tcolorbox height

### Description

Figure 2 is not formatted correctly. There is a box for the image but the image is below this box on and on top of the below text. Figure 3 is similar.

### (Optional:) Please add any files, screenshots, or other information here.


### (Required) What is this issue most closely related to? Select one.

Figures

### Internal issue ID

a81d1afc-c5c9-4a65-a72e-83e69fdbd8f3

### Paper URL

https://arxiv.org/html/2401.08702v2

### Browser

Chrome/131.0.0.0

### Device Type

Laptop

Open the paper URL in Chrome and inspect Figures 2 and 3 alongside the attached screenshots. Start by tracing how the rendered paper handles the tcolorbox figure containers; done means each image stays inside its box and does not overlap the following text.

Written by the indexing model from the issue text.