Add stages functionality