Use querySelector instead of getElementById