Added .use() to stage function