Add jQuery functionality