Added weapon functionality