Added grid reconstruct, but still have some issue