Need to add unit testing to the package

[beng1290]

We should build unit tests into the package.