added unit_test.py for unit testing