update simulate_transits.py