test.py kinda works woohooo
Ben's code is better than mine.