Add document ready handler