Adding vertical line after column in matrix table in simple layout

Hi,

I had the following code working in JavaScript to add a vertical line to the right side of the second column in a Matrix table question:

Qualtrics.SurveyEngine.addOnload(function() {
    var q = jQuery("#"+this.questionId);
    var table = q.find('table'); // Locate the table within the question

    if (table.length > 0) {
        var secondColumnHeaders = table.find('tr th:nth-child(2), tr td:nth-child(2)'); // Selecting the second column headers and cells
        secondColumnHeaders.css("border-right", "3px solid #BBBBBB"); // Adding a right border to the second column headers and cells
    }
});

However, I had to change the layout of the survey to Simple, and the code does not work anymore. How could it be solved?

Thanks in advance