Updated tool schema