Fix cross_sections.py