完善代码功能。