Update quickstart.py

Updating imports to be consistent with later parts of the documentation.