Deprecation fixes

Fixes deprecation warnings from using old @functorize macro and update jupyter notbooks and plotting examples in doc directory.