代码整理。修改几个小bug.