add stage support