已合并代码，改进功能

已经merge啦！现在应该会好一些了