Updated to work with weapons.