Updated stage class.