hastily committing test.py JUST IN CASE