Add stage diagram