oops changed something for testing by accident | also added main training script