버그 수정. 코드 정리