Improve how we traverse the DOM

I think we need to eventually do the following across the whole codebase:

1. use querySelector when we just need the first match
2. use node.querySelectorAll instead of document
3. always cache the result to a var

These might be micro-optimizations, but they should definitely help with large DOMs.