修改了实现方式

可以更加简洁的实现