adding train script and working on it, for soil_risk