Add test setup with simple example