Added train scripts