revert changing cost function
Merge branch 'run2' of https://github.com/chayut-t/grid_cells into run2