気になる箇所を変更してみました

機能的には追加してませんが、ソースで気になった部分を
修正してみました。

本当はイケてる機能の実装も考えてみたんですが、
結構時間も経ってしまったので、一旦pull request送ります。