修正Issue#4和#5

具体修改部分可以看Issue中相关回复。