modify function and readme