Add unit tests to build