Added some tests to test.py
More Python changes