Update optimal_path.py