Add unit tests with sources