Move logic to model and simplify.

軽くリファクタリングしてみたからもしよかったら(^-^)