add a small test to demo