Still trying on test.py