Add simple test for demonstartion