Who can be included under “punishment of the grave” in the relevant sense?

Punishment and Bliss in the Grave, and Questioning by Munkar and Nakir