Use jQuery instead of document for consistency

[amritghimire]

Since, we are already using jquery, We can use jquery selector to get and check whether the element exists or not.