Change registerToolMod to use ToolMod