Add setTool() function