add weapon to Unit class