Change document ready function