Whose punishment in graves is mentioned in Abu Ayyub’s report?

Recording Angels, the Angel of Death, and Questioning and Punishment in the Grave