Add initialize_context_models()