Add a better test for stack example