correct .querySelector & remove unnecessary class