Added document ready function