Updating argument types for stencil example.