기능 개선 및 코드 정리