how to integrate iframe and js on Divi?

Hello all,

I 'm new on Divi and here. I 'm looking for a solution to integrate some iframe and JS code in a Divi page. I taste with the code module but it doesn't work.

Here the code to integrate:

<script type="text/javascript">
var iframe = document.querySelector('.iframe-class');
iframe.addEventListener("load", function() {
var innerDoc = iframe.contentDocument || iframe.contentWindow.document;
var iframeHeight = innerDoc.querySelector('.container-narrow');
iframe.height = iframeHeight.clientHeight;
iframe.style.height = iframeHeight.clientHeight + "px";
});
</script>


Please Help.

Thanks