fixing the comment blocks in the ipynb