Adding sample unit tests