some corrections + tests.py file