Polish the code a little bit

d牛开始学Ruby啦？Hip hip hurray! 不过话说CS169难道是你们的课程？竟然有教Ruby的课程，真好啊...
啊废话讲太多了。今天逛的时候碰巧看到了，就帮着改了一下。没有详细测试，不过应该不至于有bug吧...