Update index.js

Don't wrap functions, only generators