add changes in integration code