Python Code Updates

Python Code Updates