added train test