Updated gene_ko_mo.ipynb notebook