Tidying up lib/iframe/index.js