로직을 모델로 변경.. 하지만 아직 문제 있음.