Added check to Area Infested, small tweaks