整理代码，添加界面功能