Fix retrieving data example

A few imports needed to be corrected. I tested the changes and it ran successfully!