add train.py
update grid.py