Switched DOM queries to jQuery