Since the tab is now called training, it would be nice to change the variable names from Display to Training everywhere.