TEST: Run testing on a 8 block loop layout, added axis labels to train diagrams