Fix saving / loading

fix saving / loading the grid