added train shell