updated ipnb with new tests