Update test.py

adding a couple functions