now saves computational grid and new flatten coordinates