いろいろ変更してます。

cp.pyの方に変更箇所書きました。
結局pull request２つ投げてます、すいません！