Update chroma_extract.py