Minor fixes to improve example