Can now restrict data based on whether it's training or test.