not needed
return np.arrays instead of lists
Include comments, add saving grids