Added new test functions to test.py.