Feature/make proper package

Changes the code into a python package that can be pip installed.

We still need to update the README etc.