Added test data and started unit_tests.py