Update to run on master

Resolves #4 and makes sure the package runs on master