Add helper to load context