又改了点代码。。。
增加controller