Add set_proxy support