Fix undo, fully working now
New gradient