Finding first code cell in a Jupyter notebook with Javascript

findFirstCodeCell.js

window.findFirstCodeCell = function findFirstCodeCell() {
  var cells = Jupyter.notebook.get_cells();
    for (var i = 0; i < cells.length; i++) { 
        if (cells[i]["cell_type"] == "code" ){
            return i;
        }
    }
}