Update context.go