Make it into a proper python module