Use jQuery instead of getElementById