Adding code for weapons