plot_design function

Plot stepped wedge design