Adding in training test