Break diagram out for each demo stage