Add tool to update tokens