refactor for tool belt