Add context import to examples that use "context"

Add context import to examples that use "context" #1584