Added the unit tests to the build