Update key_rate.py