Added basic tests, setup.py