Add imports and use the load_config function.