Add more tests for the stack