Add proxy logic