added some tests in tests.py