Added train file and class