Update test.py
Merge pull request #4 from Vidya3/test/prod

Update test.py