Adding unit tests to the build