modify function

modify function