Updated test.py with more examples