I don't know how I've missed out on contextlib all this time. I'll update that section.