Add set_config method