Add load_config function