Added the integration code