Add updateStack function