Last tweak to example