You can do this with keyboard shortcuts (but we should probably add buttons as well)...

Undo: Command + Z (or Ctrl + Z on windows)

Redo: Command + Shift + Z (or Ctrl + Shift + Z on windows)