putting code in package, adding setup.py