test.py. probably works maybe