Add stages to import