Added sample unit test