Use correct API to add tool.
Add missing import