Added undo functionality to the game screen