From a37b30f06d16dfb556c542102280c309157f22b4 Mon Sep 17 00:00:00 2001
From: Abe Levitan <abe@Rhyolite.local>
Date: Wed, 8 Jun 2022 13:33:17 -0700
Subject: [PATCH] And update the readme

---
 README.md | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index 7e12b13..2da2d8c 100644
--- a/README.md
+++ b/README.md
@@ -5,11 +5,11 @@ CDTools is a python library for ptychography and CDI reconstructions, using an A
 ```python
 # imports
 from matplotlib import pyplot as plt
-from cdtools.datasets import Ptycho_2D_Dataset
+from cdtools.datasets import Ptycho2DDataset
 from cdtools.models import FancyPtycho
 
 # Load the file
-dataset = Ptycho_2D_Dataset.from_cxi('ptycho_data.cxi')
+dataset = Ptycho2DDataset.from_cxi('ptycho_data.cxi')
 
 # Generate a model from the data
 model = FancyPtycho.from_dataset(dataset)