adding train as_json