修改bug，完善程序