修正bug，功能初步完成