Add weapon editing functionality.