And fix the final example so that I don't overwrite the collocated number datasets - they're needed for the exercise