Adding stage class