Added some weapon code