A few tweaks to include the red blocks and the beam collisions