made changes to add stages