add weapon to code