Unit testing

Adding the initial unit tests to the repo.