Adjust figure headings for screenshots

Added "figure 3" and "figure 4" to maintain consistency with use-cases