Update test.py
Merge pull request #1 from joshbeitler/master

Update test.py