enh: updated notebook with solution
enh: updated python file