Add sample unit test