Updated test.py to use schema