Add code to a document.ready