How do I make it dig 3x3 and not fall on the lava?

This is how you can tell it's gonna be a really good issue

Uh, baritone doesn't commit suicide? Just tell it to dig a 3x3

It doesn't fall into the lava.