Add checks for death cause