minor edits to get example running again