Updated selector if using pure JS