just made a quick change to use querySelector