Update diseasome_code.py