Save context

Api to update context
Remove context_layers to context(circular dependency need to be fixed first)