diff --git a/code/python/example.py b/code/python/example.py
index 1ddf7d8..917713e 100644
--- a/code/python/example.py
+++ b/code/python/example.py
@@ -1,22 +1,20 @@
-# pip3 install neo4j-driver
+# pip3 install neo4j
 # python3 example.py
 
 from neo4j import GraphDatabase, basic_auth
 
-driver = GraphDatabase.driver(
-  "neo4j://<HOST>:<BOLTPORT>",
-  auth=basic_auth("<USERNAME>", "<PASSWORD>"))
-
 cypher_query = '''
 MATCH (l:Location {address:$address})<-[r:OCCURRED_AT]-(c:Crime)
 RETURN c.date as crimeDate
 '''
 
-with driver.session(database="neo4j") as session:
-  results = session.read_transaction(
-    lambda tx: tx.run(cypher_query,
-                      address="Piccadilly").data())
-  for record in results:
-    print(record['crimeDate'])
-
-driver.close()
+with GraphDatabase.driver(
+    "neo4j://<HOST>:<BOLTPORT>",
+    auth=("<USERNAME>", "<PASSWORD>")
+) as driver:
+    result = driver.execute_query(
+        cypher_query,
+        address="Piccadilly",
+        database_="neo4j")
+    for record in result.records:
+        print(record['crimeDate'])