Add unit test snippets