Some restructuring

This makes things a bit easier to run from root of the repository. I've also added a `setup.py`