Added simple training script