Test integration

Updated the project to include some initial unit tests and an integration test.