Use querySelector instead of getElementsByClassName