Update test.py
Merge pull request #6 from Vidya3/test/prod-pr

Update test.py