Added edit UI stuff.