Add set_socks method for agents