Update to get example working