edit UI improve 2