Weapon control upgrade