Adding undo with entire stage instead of just layers + toolbar UI