I think the one thing I would ask for as a proposal is that if a family decides that they're going to go back to work, the mother, the father, the partner says, “I'm ready to go back to work,” but six months later the grief resurfaces and there's time needed to be off, that would be applicable as well.
Maybe it's 12 weeks over a period of a year and you could take a chunk here and there if you need it.