Added type annotation to example