Fixed some small issues in the example