Fixed some issues with the example.