Fixed some issues in example