Unit test modified for stack sorting exercise