Update test.py
Merge pull request #1 from Vidya3/testPR

Update test.py