修改了一下

修改了一下，增加了通用性。