Add set_proxy method