viết gọn code này lại và loại bỏ dư thừa nhưng vẫn giữ đầy đủ chức năng