removed stages in models