Changes tested and they work

- Python adjustments for the change