Some updates to function context example