Updated test.py

These commits are quite useless really, as this is just a test script