Adding unit tests to build