Training: improve suite writing example

This improves the example program for the suite writing example.