Add option to save coloc_2d plot to file