Implementing an undo button was tricky but the use of Stack made it possible. I had to add check conditions for empty stack to prevent undo when the play-area was empty.