diff --git a/jupyter-pystata-gen3-licensed/resources/setup_licensed_notebook.py b/jupyter-pystata-gen3-licensed/resources/setup_licensed_notebook.py
index ee6875a..1af7c0e 100644
--- a/jupyter-pystata-gen3-licensed/resources/setup_licensed_notebook.py
+++ b/jupyter-pystata-gen3-licensed/resources/setup_licensed_notebook.py
@@ -20,19 +20,31 @@
 
 actions = ActionChains(browser)
 actions.pause(5)
-
+actions.perform()
+browser.save_screenshot('0.png')
 # Down-arrow to get to the second cell in the notebook
 actions.send_keys(Keys.DOWN)
 actions.pause(1)
 actions.perform()
-
+browser.save_screenshot('1.png')
+#actions.perform()
+#browser.save_screenshot('1-1.png')
 # Run the stata setup cell
 print("Ready to run stata setup cell")
 actions.key_down(Keys.SHIFT)
 actions.send_keys(Keys.ENTER)
 actions.key_up(Keys.SHIFT)
+actions.pause(5)
 actions.perform()
-actions.pause(10)
+browser.save_screenshot('2.png')
+
+#print("Ready to run stata setup cell2")
+#actions.key_down(Keys.SHIFT)
+#actions.send_keys(Keys.ENTER)
+#actions.key_up(Keys.SHIFT)
+#actions.pause(10)
+#actions.perform()
+#browser.save_screenshot('2-1.png')
 
 # Save notebook with output
 print("Ready to save notebook")
@@ -40,5 +52,8 @@
 actions.send_keys("S")
 actions.key_up(Keys.CONTROL)
 actions.pause(5)
+actions.perform()
 actions.send_keys(Keys.ENTER)
+actions.pause(1)
 actions.perform()
+browser.save_screenshot('3.png')