add stage class