Updated UMD wrapper