Added setup.py and tests.py. Fixed code in datastructures per test results.