Add support for koa