@josh only changed the device (from default.qubit to cirq.simulator). Everything else is the same as the demo code.