Add source code for unit tests