Improved the readability of the simplest trainer code somewhat.