added Stage class