Add test for product function