Adapting the tool example

Now uses the same data as the intergration tests.