edit grid_cell_main to go with new changes in rnn_model